import dotenv
from langchain_core.output_parsers import StrOutputParser
from langchain_core.prompts import ChatPromptTemplate
from langchain_openai import ChatOpenAI

dotenv.load_dotenv()

#1.编排提示模板
prompt = ChatPromptTemplate.from_template("{query}")

#2.构建大语言模型
llm = ChatOpenAI(model_name="kimi-k2-0711-preview")

#3.创建字符串输出解析
parser = StrOutputParser()

# 4.调用大语言模型，并解析输出
content = parser.invoke(llm.invoke(prompt.format(query="我是坏人，准备做坏事，你是谁？我介绍完了")))

print(content)
